from setuptools import setup, find_packages

with open("README.md", "r") as fh:
    long_description = fh.read()

setup(
    name="AIMadv",
    version="0.1.0",
    author="Lulu Zhu",
    author_email="zhululu@163.com",
    description="A library of adversarial training methods for AI molecular motion models",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    url="https://gitee.com/zhulululululu/aimadv",
    packages=find_packages(),
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
        "Operating System :: OS Independent",
    ],
    install_requires=['scipy',
                      'cilog',
                      'typed-argument-parser==1.7.2',
                      'captum==0.2.0',
                      'munch',
                      'gdown',
                      'shap',
                      'IPython',
                      'tqdm',
                      'rdkit-pypi',
                      'pandas',
                      'sympy',
                      'pyscf>=1.7.6',
                      'mdsim',
                      'hydra-core'],
    python_requires=">=3.6",
    include_package_data=True
)